package org.example.mq.mqsever.datacenter;

import ch.qos.logback.core.encoder.EchoEncoder;
import org.example.mq.MqApplication;
import org.example.mq.mqsever.core.Binding;
import org.example.mq.mqsever.core.Exchange;
import org.example.mq.mqsever.core.ExchangeType;
import org.example.mq.mqsever.core.MSGQueue;
import org.example.mq.mqsever.mapper.MetaMapper;

import java.io.File;
import java.util.List;

/**
 * Created with IntelliJ IDEA.
 * Description:做渐进主义的改良派,不做完没注意的速成派
 * User: XALEXC
 * Date: 2025-05-08
 * Time: 14:16
 */

/**
 * 通过这个类来整合数据库操作
 */
public class DataBaseManager {
    // 要做的是从Spring中拿到现成的对象
    private MetaMapper metaMapper;

    //针对数据库进行初始化
    public void init(){
        // 手动获取到metaMapper
        metaMapper= MqApplication.context.getBean(MetaMapper.class);

        if (!checkDBExists()){
            // 如果数据库不存在,就进行建库建表操作
            // 先创建一个data目录
            File dataDir=new File("./data");
            dataDir.mkdirs();
            // 创建数据表
            createTable();
            // 插入默认数据集
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成");

        }else{
            // 数据库存在了,什么都不需要做
            System.out.println("[DataBaseManager] 数据库已存在");
        }
    }

    public void deleteDB(){
        // 从这里可以看出,删除文件的时候要一层一层的删除.
        File file=new File("./data/meta.db");
        boolean ret=file.delete();
        if (ret){
            System.out.println("[DataBaseManager] 删除数据库文件成功");
        }else{
            System.out.println("[DataBaseManager] 删除数据库文件失败");
        }

        File dataDir=new File("./data");
        // 使用delete删除目录的时候,需要保证目录是空的.
        ret=dataDir.delete();
        if (ret){
            System.out.println("[DataBaseManager] 删除数据库目录成功");
        }else{
            System.out.println("[DataBaseManager] 删除数据库目录失败");
        }
    }

    private boolean checkDBExists(){
        File file=new File("./data/meta.db");
        if (file.exists()){
            return true;
        }
        return false;
    }
    // 这个方法用来建表.建库操作并不需要手动执行
    // 首次执行这里的数据库操作的时候,就会自动的创建出meta.db文件来(MyBatis帮我们完成的)
    private void createTable(){
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成");
    }
    // 给数据库表中,添加默认的数据
    // 此处主要是添加一个默认的交换机
    // RabbitMQ 里有这样的一个设定: 带有一个匿名的交换机,类型是DIRECT.
    private void createDefaultData(){
        // 构造一个默认的交换机
        Exchange exchange=new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 创建初始数据完成");
    }

//    public void shutdown() {
//        try {
//            // 关闭 PooledDataSource 的所有连接
//            metaMapper = null;
//            System.out.println("[DataBaseManager] MetaMapper 已置空");
//
//            // 如果你能获取到 SqlSessionFactory，就调用它的 close() 或清理连接池
//            // 示例（伪代码，取决于你的配置方式）:
//        /*
//        SqlSessionFactory sqlSessionFactory = MqApplication.context.getBean(SqlSessionFactory.class);
//        if (sqlSessionFactory instanceof DefaultSqlSessionFactory) {
//            ((DefaultSqlSessionFactory) sqlSessionFactory).close(); // 释放资源
//        }
//        */
//        } catch (Exception e) {
//            System.out.println("[DataBaseManager] 关闭数据库资源失败");
//            e.printStackTrace();
//        }
//    }


    // 把其他数据库的操作也在这里封装一下,就不需要每次使用都要调用metaMapper
    public void insertExchange(Exchange exchange){
        metaMapper.insertExchange(exchange);
    }

    public void insertQueue(MSGQueue queue){
        metaMapper.insertQueue(queue);
    }

    public void insertBinding(Binding binding){
        metaMapper.insertBinding(binding);
    }

    public void deleteExchange(String exchangeName){
        metaMapper.deleteExchange(exchangeName);
    }

    public void deleteQueue(String queueName){
        metaMapper.deleteQueue(queueName);
    }

    public void deleteBinding(Binding binding){
        metaMapper.deleteBinding(binding);
    }

    public List<Exchange> selectAllExchange(){
        return metaMapper.selectAllExchanges();
    }

    public List<MSGQueue> selectAllQueue(){
        return metaMapper.selectAllQueues();
    }

    public List<Binding> selectAllBinding(){
        return metaMapper.selectAllBindings();
    }
}
